cs201p assignment 1 solution